/*
 * Copyright (C) 2020 Anton Malinskiy
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.malinskiy.adam.server.stub

import assertk.assertThat
import assertk.assertions.isEqualTo
import assertk.assertions.startsWith
import io.ktor.utils.io.ByteReadChannel
import io.ktor.utils.io.readUTF8Line

class ConsoleReadChannel(private val delegate: ByteReadChannel) : ByteReadChannel by delegate {
    suspend fun receiveCommand() = delegate.readUTF8Line()!!
    suspend fun receiveAuth() = receiveCommand().apply { assertThat(this).startsWith("auth ") }.substringAfter("auth ")
    suspend fun receiveExit() = receiveCommand().apply { assertThat(this).isEqualTo("quit") }
}
